import openpyxl


class ExcelHelper(object):

    # 获取Sheet
    @staticmethod
    def get_work_sheet(file_name, sheet_name):
        work_book = openpyxl.load_workbook(file_name)
        work_sheet = work_book.get_sheet_by_name(sheet_name)
        return work_sheet

    # 获取表格中的值
    @staticmethod
    def get_val_from_sheet(sheet, src_val):
        # 遍历所有行知道找到对应的值并将该值返回
        for row in sheet.iter_rows(min_row=1, min_col=1, max_col=2):
            try:
                # print(row[0].value, row[1].value)
                # 如果取到的第一列的值是None则返回
                if row[0].value is None:
                    return None
                # 如果取到的第一列的值与传进来的值匹配则返回第二列的翻译后的字符串
                if row[0].value == src_val:
                    return row[1].value
            except:
                return None
